Cognitive training control